package Paper;

import java.util.ArrayList;

/*
    1:定义一个ClassRoom对象,包含构造方法,提供get/set方法
    班级名称,className(String 类型)	班级类型:classType(String 类型)
    班级人数:classCount(int 类型)
    2.定义测试类ClassTest,完成以下要求：
    定义方法，public static ArrayList<User> getClassCount (ArrayList<User> users)，获得班
    级人数大于50人，并且班级类型是Java的，添加到新的集合中返回。
    在main方法中完成以下功能：创建2个ClassRoom对象并赋值，并且将ClassRoom对象添加到ArrayList<ClassRoom>集合中，调用getClassCount方法并接收，遍历返回的新集合输出新集合中所有元素信息

 */
class ClassRoom{
    private String className,classType;
    private int classCount;

    //构造方法
    public ClassRoom() {
    }

    public ClassRoom(String className, String classType, int classCount) {
        this.className = className;
        this.classType = classType;
        this.classCount = classCount;
    }

    public String getClassName() {
        return className;
    }

    public void setClassName(String className) {
        this.className = className;
    }

    public String getClassType() {
        return classType;
    }

    public void setClassType(String classType) {
        this.classType = classType;
    }

    public int getClassCount() {
        return classCount;
    }

    public void setClassCount(int classCount) {
        this.classCount = classCount;
    }
}
public class ClassTest {
    public static void main(String[] args) {
        //创建2个ClassRoom对象并赋值
        // 并且将ClassRoom对象添加到ArrayList<ClassRoom>集合中
        ArrayList<ClassRoom> classRoom = new ArrayList<>();
        ClassRoom c = new ClassRoom("37期","Java",60);
        classRoom.add(c);
        c = new ClassRoom("12期","前端",60);
        // 调用getClassCount方法并接收
        ArrayList<ClassRoom> classResult = getClassCount(classRoom);
        // 遍历返回的新集合输出新集合中所有元素信息
        System.out.println("新集合中所有元素信息:"+classResult.size()+"个");
        for (int i = 0; i < classResult.size(); i++) {
            System.out.println("班级名:"+classResult.get(i).getClassName()+" 班级类型:"+classResult.get(i).getClassType()+" 班级人数:"+classResult.get(i).getClassCount());
        }
    }

    /*
    定义方法，public static ArrayList<User> getClassCount (ArrayList<User> users)
    获得班级人数大于50人，并且班级类型是Java的，添加到新的集合中返回
     */
    public static ArrayList<ClassRoom> getClassCount (ArrayList<ClassRoom> classRoom){
        ArrayList<ClassRoom> classNew = new ArrayList<>();
        for (int i = 0; i < classRoom.size(); i++) {
            if (classRoom.get(i).getClassCount() > 50 && classRoom.get(i).getClassType().equals("Java")){
                classNew.add(classRoom.get(i));
            }
        }
        return classNew;
    }
}
